from major.models_manager import chat_model
from langgraph.prebuilt import create_react_agent
import datetime

def current_time():
    """返回时间的工具"""
    return datetime.datetime.now()

model = chat_model.get_model()
# 创建Agent，包含两个工具：calculator 和 current_time
agent = create_react_agent(model,
                           tools=[current_time],
                           prompt='''你是一个时间管理AI''')


response  = agent.invoke({'messages': [{'role': 'user', 'content': '现在几点了？'}]})
print(response['messages'][-1].content)